# Author: HZT
# Date: 2023/10/20
import time
import page
from base.base import Base
from utils import click_paste_img


class WebCkHk(Base):
    # 获取余额
    def ck_hk_get_ye(self):
        return self.base_get_text(page.web_ck_hk_ye)

    # 点击回款管理
    def ck_hk_click_hkgl(self):
        self.base_click(page.web_ck_hk_hkgl)

    # 获取新增元素
    def ck_hk_get_xz(self):
        return self.base_find(page.web_ck_hk_xz)

    # 点击新增
    def ck_hk_click_xz(self):
        self.base_click(page.web_ck_hk_xz)

    # 获取新增页面中的余额值
    def ck_hk_get_xzye(self):
        return self.base_find(page.web_ck_hk_xzye).get_attribute("value")

    # 输入回款金额
    def ck_hk_input_hkje(self, text):
        self.base_input(page.web_ck_hk_hkje, text)

    # 输入银行名
    def ck_hk_input_yhm(self, text):
        self.base_input(page.web_ck_hk_yhm, text)

    # 输入姓名
    def ck_hk_input_xm(self, text):
        self.base_input(page.web_ck_hk_xm, text)

    # 输入卡号
    def ck_hk_input_kh(self, text):
        self.base_input(page.web_ck_hk_kh, text)

    # 获取二维码粘贴元素
    def ck_hk_get_ewm(self):
        return self.base_find(page.web_ck_hk_ewm)

    # 输入备注
    def ck_hk_input_bz(self, text):
        self.base_input(page.web_ck_hk_bz, text)

    # 点击提交
    def ck_hk_click_tj(self):
        self.base_click(page.web_ck_hk_tj)

    # 点击回款待处理
    def ck_hk_click_hkdcl(self):
        self.base_click(page.web_ck_hk_hkdcl)

    # 获取收款账号
    def ck_hk_get_skzh(self):
        return self.base_get_text(page.web_ck_hk_skzh)

    # 获取充值金额
    def ck_hk_get_czje(self):
        return self.base_get_text(page.web_ck_hk_czje)

    # 点击操作
    def ck_hk_click_cz(self):
        self.base_click(page.web_ck_hk_cz)

    # 点击确认完成
    def ck_hk_click_qrwc(self):
        self.base_click(page.web_ck_hk_qrwc)

    # 获取状态
    def ck_hk_get_zt(self):
        return self.base_get_text(page.web_ck_hk_zt)

    # 获取打款截图元素
    def ck_hk_get_dkjt(self):
        return self.base_find(page.web_ck_hk_dkjt)

    # 点击确认提交
    def ck_hk_click_qrtj(self):
        self.base_click(page.web_ck_hk_qrtj)

    # 点击提交OK
    def ck_hk_click_tjok(self):
        self.base_click(page.web_ck_hk_tjok)

    # 组合创建回款业务
    def ck_hk_xz(self, driver, hkje, yhm, xm, kh, filename, bz=None):
        self.ck_hk_input_hkje(hkje)
        self.ck_hk_input_yhm(yhm)
        self.ck_hk_input_xm(xm)
        self.ck_hk_input_kh(kh)
        driver.execute_script("window.scrollBy(0, 500);")
        element = self.ck_hk_get_ewm()
        click_paste_img(filename, element, driver)
        time.sleep(0.5)
        if bz:
            self.ck_hk_input_bz(bz)
        self.ck_hk_click_tj()
